#!/usr/bin/env python3

import pandas as pd
import numpy as np

# 以最近一个月月末为例，绘制所选股票范围中所有股票的总市值的分布和size因子分布
# 最近一个月月末的所有股票的总市值的分布和size因子分布
# 总市值 流通股数 * 股票现值， 这里老师已经给我们了算好的数据


# print(size_factor_df2)
class SizeFactor:
    def __init__(self, qd):
        self.qd = qd
        self.df = qd.equity_data.query('E_RPT_PERIOD>"20210830"')
        self.df2 = pd.merge(self.df, qd.ind1k8, how='inner', on=["STOCK_CODE", "STOCK_CODE"])
        self.extreme_stock = self.df2.query('TOTAL_EQUITY_PARENT <= 0')
        print(self.extreme_stock)
        self.dat = pd.Series(np.log(self.df2.query('TOTAL_EQUITY_PARENT > 0')['TOTAL_EQUITY_PARENT']))

    def plot(self):
        print(self.dat)
        # dat.hist(bins=20, )
        # np.histogram(dat, range=(dat.min(), dat.max()), bins=20)
